#pragma once 
#include <iostream>
#include <unistd.h>
#include <queue>
#include <pthread.h>

const int gmaxcap = 5;

template <class T>
class BlockQueue
{
public:
    BlockQueue(const int &maxcap = gmaxcap): _maxcap(maxcap)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_pcond, nullptr);
        pthread_cond_init(&_ccond, nullptr);
    }
    void push(const T &in)
    {
        pthread_mutex_lock(&_mutex);
        // 充当条件判断的语法必须是while, 不能用if
        while(is_full()) // 为了排除伪唤起的可能, 例如:
        // 只有一个数据被消耗, 但是一次唤醒多个线程
        {
            // pthread_cond_wait 的第二个参数必须是被正在被使用的互斥锁
            // pthread_cond_wait 内部会以原子操作来释放锁, 并将自己挂起
            // pthread_cond_wait 在被唤醒时, 会自动获取传入的锁, 直到成功
            pthread_cond_wait(&_pcond, &_mutex);
        }
        _q.push(in);
        // pthread_cond_signal 可以放在临界区内部, 也可以放在外部
        pthread_cond_signal(&_ccond);
        pthread_mutex_unlock(&_mutex);
    }
    void pop(T *out)
    {
        pthread_mutex_lock(&_mutex);
        while(is_empty()) // bug?
        {
            pthread_cond_wait(&_ccond, &_mutex);
        }
        *out = _q.front();
        _q.pop();
        pthread_cond_signal(&_pcond);
        pthread_mutex_unlock(&_mutex);
    }
    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_pcond);
        pthread_cond_destroy(&_ccond);
    }
private:
    bool is_empty()
    {
        return _q.empty();
    }
    bool is_full()
    {
        return _q.size() == _maxcap;
    }
private:
    std::queue<T> _q;
    int _maxcap; // 队列的最大上限
    pthread_mutex_t _mutex;
    pthread_cond_t _pcond; // 生产者的条件变量
    pthread_cond_t _ccond; // 消费者的条件变量
};